package com.ikingtech.platform.service.system.role.service;

import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.ikingtech.framework.sdk.menu.api.MenuRoleApi;
import com.ikingtech.platform.service.system.role.entity.RoleMenuDO;
import lombok.RequiredArgsConstructor;

import java.util.Collection;
import java.util.List;

/**
 * @author tie yan
 */
@RequiredArgsConstructor
public class MenuRoleService implements MenuRoleApi {

    private final RoleMenuRepository roleMenuService;

    @Override
    public void removeRoleMenu(Collection<String> menuIds) {
        this.roleMenuService.remove(Wrappers.<RoleMenuDO>lambdaQuery()
                .in(RoleMenuDO::getMenuId, menuIds));
    }

    @Override
    public List<String> loadMenuId(String roleId) {
        return this.roleMenuService.listObjs(Wrappers.<RoleMenuDO>lambdaQuery()
                .select(RoleMenuDO::getMenuId)
                .eq(RoleMenuDO::getRoleId, roleId), String.class::cast);
    }

    @Override
    public List<String> loadMenuId(List<String> roleIds) {
        return this.roleMenuService.listObjs(Wrappers.<RoleMenuDO>lambdaQuery()
                .select(RoleMenuDO::getMenuId)
                .in(RoleMenuDO::getRoleId, roleIds), String.class::cast);
    }
}
